home *** CD-ROM | disk | FTP | other *** search
/ Aminet 30 / Aminet 30 (1999)(Schatztruhe)[!][Apr 1999].iso / Aminet / gfx / show / mpeg2decodeWOS.lha / mpeg2decode / src / spatscal.c < prev    next >
C/C++ Source or Header  |  1999-02-23  |  10KB  |  358 lines
  1.  
  2. #include <stdio.h>
  3. #include "config.h"
  4. #include "global.h"
  5.  
  6. /* private prototypes */
  7. static void Read_Lower_Layer_Component_Framewise _ANSI_ARGS_((int comp, int lw, int lh));
  8. static void Read_Lower_Layer_Component_Fieldwise _ANSI_ARGS_((int comp, int lw, int lh));
  9. static void Make_Spatial_Prediction_Frame _ANSI_ARGS_((int progressive_frame,
  10.   int llprogressive_frame, unsigned char *fld0, unsigned char *fld1, 
  11.   short *tmp, unsigned char *dst, int llx0, int lly0, int llw, int llh, 
  12.   int horizontal_size, int vertical_size, int vm, int vn, int hm, int hn, 
  13.   int aperture));
  14. static void Deinterlace _ANSI_ARGS_((unsigned char *fld0, unsigned char *fld1,
  15.   int j0, int lx, int ly, int aperture));
  16. static void Subsample_Vertical _ANSI_ARGS_((unsigned char *s, short *d,
  17.   int lx, int lys, int lyd, int m, int n, int j0, int dj));
  18. static void Subsample_Horizontal _ANSI_ARGS_((short *s, unsigned char *d,
  19.   int x0, int lx, int lxs, int lxd, int ly, int m, int n));
  20.  
  21.  
  22.  
  23. /* get reference frame */
  24. void Spatial_Prediction()
  25. {
  26.   
  27.   if(Frame_Store_Flag)
  28.   {
  29.     Read_Lower_Layer_Component_Framewise(0,lower_layer_prediction_horizontal_size, 
  30.       lower_layer_prediction_vertical_size);      /* Y */
  31.     Read_Lower_Layer_Component_Framewise(1,lower_layer_prediction_horizontal_size>>1,
  32.       lower_layer_prediction_vertical_size>>1);   /* Cb ("U") */
  33.     Read_Lower_Layer_Component_Framewise(2,lower_layer_prediction_horizontal_size>>1,
  34.       lower_layer_prediction_vertical_size>>1);   /* Cr ("V") */
  35.   }
  36.   else
  37.   {
  38.     Read_Lower_Layer_Component_Fieldwise(0,lower_layer_prediction_horizontal_size, 
  39.       lower_layer_prediction_vertical_size);      /* Y */
  40.     Read_Lower_Layer_Component_Fieldwise(1,lower_layer_prediction_horizontal_size>>1,
  41.       lower_layer_prediction_vertical_size>>1);   /* Cb ("U") */
  42.     Read_Lower_Layer_Component_Fieldwise(2,lower_layer_prediction_horizontal_size>>1,
  43.       lower_layer_prediction_vertical_size>>1);   /* Cr ("V") */
  44.   }
  45.  
  46.  
  47.   Make_Spatial_Prediction_Frame  /* Y */
  48.     (progressive_frame,lower_layer_progressive_frame,llframe0[0],llframe1[0],
  49.      lltmp,current_frame[0],lower_layer_horizontal_offset,
  50.      lower_layer_vertical_offset,
  51.      lower_layer_prediction_horizontal_size,
  52.      lower_layer_prediction_vertical_size,
  53.      horizontal_size,vertical_size,vertical_subsampling_factor_m,
  54.      vertical_subsampling_factor_n,horizontal_subsampling_factor_m,
  55.      horizontal_subsampling_factor_n,
  56.      picture_structure!=FRAME_PICTURE); /* this changed from CD to DIS */
  57.  
  58.   Make_Spatial_Prediction_Frame  /* Cb */
  59.     (progressive_frame,lower_layer_progressive_frame,llframe0[1],llframe1[1],
  60.      lltmp,current_frame[1],lower_layer_horizontal_offset/2,
  61.      lower_layer_vertical_offset/2,
  62.      lower_layer_prediction_horizontal_size>>1,
  63.      lower_layer_prediction_vertical_size>>1,
  64.      horizontal_size>>1,vertical_size>>1,vertical_subsampling_factor_m,
  65.      vertical_subsampling_factor_n,horizontal_subsampling_factor_m,
  66.      horizontal_subsampling_factor_n,1);
  67.  
  68.   Make_Spatial_Prediction_Frame  /* Cr */
  69.     (progressive_frame,lower_layer_progressive_frame,llframe0[2],llframe1[2],
  70.      lltmp,current_frame[2],lower_layer_horizontal_offset/2,
  71.      lower_layer_vertical_offset/2,
  72.      lower_layer_prediction_horizontal_size>>1,
  73.      lower_layer_prediction_vertical_size>>1,
  74.      horizontal_size>>1,vertical_size>>1,vertical_subsampling_factor_m,
  75.      vertical_subsampling_factor_n,horizontal_subsampling_factor_m,
  76.      horizontal_subsampling_factor_n,1);
  77.  
  78. }
  79.  
  80. #ifdef STORM
  81. static void Read_Lower_Layer_Component_Framewise(int comp,int lw,int lh)
  82. #else
  83. static void Read_Lower_Layer_Component_Framewise(comp,lw,lh)
  84.      int comp;
  85.      int lw, lh;
  86. #endif
  87. {
  88.   FILE *fd;
  89.   char fname[256];
  90.   char ext[3][3] = {".Y",".U",".V"}; 
  91. /*  char *ext = {".Y",".U",".V"}; */
  92.   int i,j;
  93.  
  94.   sprintf(fname,Lower_Layer_Picture_Filename,True_Framenum);
  95.   strcat(fname,ext[comp]);
  96. #ifdef VERBOSE
  97.   if (Verbose_Flag>1)
  98.     printf("reading %s\n",fname);
  99. #endif VERBOSE
  100.   fd=fopen(fname,"rb");
  101.   if (fd==NULL) exit(-1);
  102.   for (j=0; j<lh; j++) {
  103.      for (i=0; i<lw; i++)
  104.        llframe0[comp][lw*j+i]=getc(fd);
  105.      if (! lower_layer_progressive_frame) {
  106.     j++;
  107.     for (i=0; i<lw; i++)
  108.       llframe1[comp][lw*j+i]=getc(fd);
  109.      }
  110.   }
  111.   fclose(fd);
  112. }
  113.  
  114.  
  115. #ifdef STORM
  116. static void Read_Lower_Layer_Component_Fieldwise(int comp,int lw,int lh)
  117. #else
  118. static void Read_Lower_Layer_Component_Fieldwise(comp,lw,lh)
  119.      int comp;
  120.      int lw, lh;
  121. #endif
  122. {
  123.   FILE *fd;
  124.   char fname[256];
  125.   char ext[3][3] = {".Y",".U",".V"}; 
  126. /*  char *ext = {".Y",".U",".V"}; */
  127.   int i,j;
  128.  
  129.   sprintf(fname,Lower_Layer_Picture_Filename,True_Framenum,lower_layer_progressive_frame ? 'f':'a');
  130.   strcat(fname,ext[comp]);
  131. #ifdef VERBOSE
  132.   if (Verbose_Flag>1)
  133.     printf("reading %s\n",fname);
  134. #endif VERBOSE
  135.   fd=fopen(fname,"rb");
  136.   if (fd==NULL) exit(-1);
  137.   for (j=0; j<lh; j+=lower_layer_progressive_frame?1:2)
  138.     for (i=0; i<lw; i++)
  139.       llframe0[comp][lw*j+i]=getc(fd);
  140.   fclose(fd);
  141.  
  142.   if (! lower_layer_progressive_frame) {
  143.     sprintf(fname,Lower_Layer_Picture_Filename,True_Framenum,'b');
  144.     strcat(fname,ext[comp]);
  145. #ifdef VERBOSE
  146.     if (Verbose_Flag>1)
  147.       printf("reading %s\n",fname);
  148. #endif VERBOSE
  149.     fd=fopen(fname,"rb");
  150.     if (fd==NULL) exit(-1);
  151.     for (j=1; j<lh; j+=2)
  152.       for (i=0; i<lw; i++)
  153.         llframe1[comp][lw*j+i]=getc(fd);
  154.     fclose(fd);
  155.   }
  156. }
  157.  
  158.  
  159. /* form spatial prediction */
  160. #ifdef STORM
  161. static void Make_Spatial_Prediction_Frame(int progressive_frame,
  162.   int llprogressive_frame,unsigned char *fld0,unsigned char *fld1,short *tmp,unsigned char *dst,int llx0,int lly0,int llw,int llh,int horizontal_size,
  163.   int vertical_size,int vm,int vn,int hm,int hn,int aperture)
  164. #else
  165. static void Make_Spatial_Prediction_Frame(progressive_frame,
  166.   llprogressive_frame,fld0,fld1,tmp,dst,llx0,lly0,llw,llh,horizontal_size,
  167.   vertical_size,vm,vn,hm,hn,aperture)
  168. int progressive_frame,llprogressive_frame;
  169. unsigned char *fld0,*fld1;
  170. short *tmp;
  171. unsigned char *dst;
  172. int llx0,lly0,llw,llh,horizontal_size,vertical_size,vm,vn,hm,hn,aperture;
  173. #endif
  174. {
  175.   int w, h, x0, llw2, llh2;
  176.  
  177.   llw2 = (llw*hn)/hm;
  178.   llh2 = (llh*vn)/vm;
  179.  
  180.   if (llprogressive_frame)
  181.   {
  182.     /* progressive -> progressive / interlaced */
  183.     Subsample_Vertical(fld0,tmp,llw,llh,llh2,vm,vn,0,1);
  184.   }
  185.   else if (progressive_frame)
  186.   {
  187.     /* interlaced -> progressive */
  188.     if (lower_layer_deinterlaced_field_select)
  189.     {
  190.       Deinterlace(fld1,fld0,0,llw,llh,aperture);
  191.       Subsample_Vertical(fld1,tmp,llw,llh,llh2,vm,vn,0,1);
  192.     }
  193.     else
  194.     {
  195.       Deinterlace(fld0,fld1,1,llw,llh,aperture);
  196.       Subsample_Vertical(fld0,tmp,llw,llh,llh2,vm,vn,0,1);
  197.     }
  198.   }
  199.   else
  200.   {
  201.     /* interlaced -> interlaced */
  202.     Deinterlace(fld0,fld1,1,llw,llh,aperture);
  203.     Deinterlace(fld1,fld0,0,llw,llh,aperture);
  204.     Subsample_Vertical(fld0,tmp,llw,llh,llh2,vm,vn,0,2);
  205.     Subsample_Vertical(fld1,tmp,llw,llh,llh2,vm,vn,1,2);
  206.   }
  207.  
  208.     /* vertical limits */
  209.     if (lly0<0)
  210.     {
  211.       tmp-= llw*lly0;
  212.       llh2+= lly0;
  213.       if (llh2<0)
  214.         llh2 = 0;
  215.       h = (vertical_size<llh2) ? vertical_size : llh2;
  216.     }
  217.     else
  218.     {
  219.       dst+= horizontal_size*lly0;
  220.       h= vertical_size - lly0;
  221.       if (h>llh2)
  222.         h = llh2;
  223.     }
  224.  
  225.     /* horizontal limits */
  226.     if (llx0<0)
  227.     {
  228.       x0 = -llx0;
  229.       llw2+= llx0;
  230.       if (llw2<0)
  231.         llw2 = 0;
  232.       w = (horizontal_size<llw2) ? horizontal_size : llw2;
  233.     }
  234.     else
  235.     {
  236.       dst+= llx0;
  237.       x0 = 0;
  238.       w = horizontal_size - llx0;
  239.       if (w>llw2)
  240.         w = llw2;
  241.     }
  242.   
  243.   Subsample_Horizontal(tmp,dst,x0,w,llw,horizontal_size,h,hm,hn);
  244. }
  245.  
  246. /* deinterlace one field (interpolate opposite parity samples)
  247.  *
  248.  * deinterlacing is done in-place: if j0=1, fld0 contains the input field in
  249.  * its even lines and the odd lines are interpolated by this routine
  250.  * if j0=0, the input field is in the odd lines and the even lines are
  251.  * interpolated
  252.  *
  253.  * fld0: field to be deinterlaced
  254.  * fld1: other field (referenced by the two field aperture filter)
  255.  * j0:   0: interpolate even (top) lines, 1: interpolate odd (bottom) lines
  256.  * lx:   width of fld0 and fld1
  257.  * ly:   height of the deinterlaced field (has to be even)
  258.  * aperture: 1: use one field aperture filter (two field otherwise)
  259.  */
  260. #ifdef STORM
  261. static void Deinterlace(unsigned char *fld0,unsigned char *fld1,int j0,int lx,int ly,int aperture)
  262. #else
  263. static void Deinterlace(fld0,fld1,j0,lx,ly,aperture)
  264. unsigned char *fld0,*fld1;
  265. int j0,lx,ly; /* ly has to be even */
  266. int aperture;
  267. #endif
  268. {
  269.   int i,j,v;
  270.   unsigned char *p0, *p0m1, *p0p1, *p1, *p1m2, *p1p2;
  271.  
  272.   /* deinterlace one field */
  273.   for (j=j0; j<ly; j+=2)
  274.   {
  275.     p0 = fld0+lx*j;
  276.     p0m1 = (j==0)    ? p0+lx : p0-lx;
  277.     p0p1 = (j==ly-1) ? p0-lx : p0+lx;
  278.  
  279.     if (aperture)
  280.       for (i=0; i<lx; i++)
  281.         p0[i] = (unsigned int)(p0m1[i] + p0p1[i] + 1)>>1;
  282.     else
  283.     {
  284.       p1 = fld1 + lx*j;
  285.       p1m2 = (j<2)     ? p1 : p1-2*lx;
  286.       p1p2 = (j>=ly-2) ? p1 : p1+2*lx;
  287.       for (i=0; i<lx; i++)
  288.       {
  289.         v = 8*(p0m1[i]+p0p1[i]) + 2*p1[i] - p1m2[i] - p1p2[i];
  290.         p0[i] = Clip[(v + ((v>=0) ? 8 : 7))>>4];
  291.       }
  292.     }
  293.   }
  294. }
  295.  
  296. /* vertical resampling */
  297. #ifdef STORM
  298. static void Subsample_Vertical(unsigned char *s,short *d,int lx,int lys,int lyd,int m,int n,int j0,int dj)
  299. #else
  300. static void Subsample_Vertical(s,d,lx,lys,lyd,m,n,j0,dj)
  301. unsigned char *s;
  302. short *d;
  303. int lx, lys, lyd, m, n, j0, dj;
  304. #endif
  305. {
  306.   int i, j, c1, c2, jd;
  307.   unsigned char *s1, *s2;
  308.   short *d1;
  309.  
  310.   for (j=j0; j<lyd; j+=dj)
  311.   {
  312.     d1 = d + lx*j;
  313.     jd = (j*m)/n;
  314.     s1 = s + lx*jd;
  315.     s2 = (jd<lys-1)? s1+lx : s1;
  316.     c2 = (16*((j*m)%n) + (n>>1))/n;
  317.     c1 = 16 - c2;
  318.     for (i=0; i<lx; i++)
  319.       d1[i] = c1*s1[i] + c2*s2[i];
  320.   }
  321. }
  322.  
  323. /* horizontal resampling */
  324. #ifdef STORM
  325. static void Subsample_Horizontal(short *s,unsigned char *d,int x0,int lx,int lxs,int lxd,int ly,int m,int n)
  326. #else
  327. static void Subsample_Horizontal(s,d,x0,lx,lxs,lxd,ly,m,n)
  328. short *s;
  329. unsigned char *d;
  330. int x0, lx, lxs, lxd, ly, m, n;
  331. #endif
  332. {
  333.   int i, i1, j, id, c1, c2, v;
  334.   short *s1, *s2;
  335.   unsigned char *d1;
  336.  
  337.   for (i1=0; i1<lx; i1++)
  338.   {
  339.     d1 = d + i1;
  340.     i = x0 + i1;
  341.     id = (i*m)/n;
  342.     s1 = s+id;
  343.     s2 = (id<lxs-1) ? s1+1 : s1;
  344.     c2 = (16*((i*m)%n) + (n>>1))/n;
  345.     c1 = 16 - c2;
  346.     for (j=0; j<ly; j++)
  347.     {
  348.       v = c1*(*s1) + c2*(*s2);
  349.       *d1 = (v + ((v>=0) ? 128 : 127))>>8;
  350.       d1+= lxd;
  351.       s1+= lxs;
  352.       s2+= lxs;
  353.     }
  354.   }
  355. }
  356.  
  357.  
  358.